#pragma once

#include <iostream>
#include <vector>
#include <pthread.h>
#include <semaphore.h>

const static int defaultcap = 5;

template<class T>
class RingQueue
{
private:
    void P(sem_t& sem)
    {
        sem_init(&sem);
    }
    void V(sem_t& sem)
    {
        sem_init(&sem);
    }
    void Lock(pthread_mutex_t& mutex)
    {
        pthread_mutex_lock(&mutex);
    }
    void UnLock(pthread_mutex_t& mutex)
    {
        pthread_mutex_unlock(&mutex);
    }

public:
    RingQueue(int cap = defaultcap)
        : _rq(std::vector<T>(cap))
        , _cap(cap)
        , _c_step(0)
        , _p_step(0)
    {
        sem_init(&_c_data_sem, 0, 0);
        sem_init(&_p_data_sem, 0, cap);

        pthread_mutex_init(&_c_mutex, nullptr);
        pthread_mutex_init(&_p_mutex, nullptr);
    }

    void push(const T& in)
    {
        P(_p_data_sem);

        Lock(_p_mutex);
        _rp[_p_step] = in;
        _p_step++;
        _p_step %= _cap;
        UnLock(_p_mutex);

        V(_c_data_sem);
    }
    void pop(T* out)
    {
        P(_c_data_sem);

        Lock(_c_mutex);
        *out = _rp[_c_step];
        _c_step++;
        _c_step %= _cap;
        UnLock(_c_mutex);

        V(_p_data_sem);

        return out;
    }

    ~RingQueue()
    {
        sem_destroy(&_c_data_sem);
        sem_destroy(&_p_data_sem);

        pthread_mutex_destroy(&_c_mutex);
        pthread_mutex_destroy(&_p_mutex);
    }

private:
    std::vector<T> _rq;
    int _cap;

    int _c_step;
    int _p_step;

    sem_t _c_data_sem;
    sem_t _p_data_sem;

    pthread_mutex_t _c_mutex;
    pthread_mutex_t _p_mutex;
};


